# 需求: 查找当前目录下面的所有的 .py文件和 .txt文件

path = r"C:\python06\day11\代码"
#
import os
def get_files(path):
    # 判断传入的路径是否合法
    if not os.path.exists(path):
        print("路径不存在!")
        return
    file_list = os.listdir()
    print(file_list)
    for file in file_list:
        if file.endswith(".py") or file.endswith(".txt"):
            print(file)

# get_files(path)


# 需求: 使用递归的方式遍历newdir文件夹下面的所有文件和文件夹
path1 = r"C:\python06\day11\笔记\newdir"
def search_file(path1):
    if not os.path.exists(path1):
        print("路径不存在")
        return
    file_list = os.listdir(path1)
    # print(file_list)
    for file in file_list:
        # print(file)
        # 获取文件或者文件夹的绝对路径
        file_path = os.path.join(path1,file)
        print(file_path)
        # 判断路径是否是文件
        if os.path.isfile(file_path):
            print("----",file,"是文件")
        # 否则是文件夹
        else:
            print("----",file,"是文件夹")
            # 递归:
            search_file(file_path)
search_file(path1)